pytorch 安装与jupyter notebook使用 |
您所在的位置：网站首页 › jupyter notebook如何下载 › pytorch 安装与jupyter notebook使用 |
pytorch 安装与jupyter notebook使用
|
1. 前言
最近准备开始学习pytorch，所以准备每天学习一个pytorch的函数，并且会将开发过程中遇到的一些困难以及解决办法全部展示出来，方便自己以后查看，也希望和有这方面意向的人多交流和学习。 2. 安装pytorch由于一些人（包括我）也是刚入门学习pytorch，所以我们从pytorch的安装开始讲起。 pytorch的安装大家不要去各种神奇的地方看安装攻略，它们写的很多东西都比较老了，直接点击以下官网链接pytorch 官网。向下滑动看到如下所示界面： 然后点击自己想要使用的安装方式，第一个选项：安装版本，建议安装(1.7.0)，第二个看自己的电脑是什么就选什么。第三个注意，conda方式是当你使用anaconda才可以使用conda选项，当然我也推荐使用conda方式，因为anaconda管理python很方便，anaconda安装使用我以前写的博客或者网上找教程就可以。pip是python自带安装方式，如果你不选用anaconda，那么用pip也是可以的。语言选择pythoncuda是看你需不需要使用GPU加速，如果你要使用，那么就选择，cuda建议使用10.2，不过如果你同时安装了tensorflow的话，为了使你的tensorflow还可以继续使用GPU加速，建议使用10.1（因为TensorFlow目前最多支持到10.1）。然后如果你从来没哟使用过GPU加速，光安装cuda是不够的，我们还需要安装cudnn，这个东西是让你的显卡具备科学计算的能力，安装教程可以百度搜索，翻我以前的也可以，就是比较老了，还是建议百度。然后在你的anaconda prompt里面输入run the command里面的命令即可。 安装过程比较慢，请耐心等待。
安装好以后，本次大部分使用jupyter notebook来进行演示，当然后面可能会用vscode（因为本人转向了vscode，而且真的好用，有时间我出教程如何使用vscode来编写，当然pycharm的优点是便于调试；缺点是启动慢，占内存） 3. 使用jupyter notebook 电脑点击启动按键，输入jupyter notebook即可 选择第一个，点击后看到如下界面：  点击new，选择python3看到如下界面：  然后开始使用，输入代码，首先测试pytorch是否安装成功： 安装cpu输入
import torch
按(shift + enter)运行，没报错即为正常。 安装GPU版本输入： import torch print(torch.cuda.is_available()) 输出为True即为正常，输出为False说明cuda安装失败。如果报no module named "torch"则是因为pytorch没有安装成功，如果显示缺少了一个dll文件，那么是因为C++库没有安装，点击以下官方链接visual C++下载，进去后下载x86与x64文件，依次安装x86与x64文件，然后重启电脑即可解决： 至此pytorch安装成功，然后开始pytorch学习之旅。
|
【本文地址】
今日新闻 |
推荐新闻 |